package leetcode.day;

// 1716. 计算力扣银行的钱
public class TotalMoney {
    public static void main(String[] args) {
        System.out.println(totalMoney(20));
    }

    // 等差数列，把第一周的1+...7看做一个整体
    public static int totalMoney(int n) {
        int sum = 0;
        int a1 = (1 + 7) * 7 / 2;
        // 等差数列求和公式 Sn=n*a1+n(n-1)d/2
        int sn = (n / 7) * a1 + (n / 7) * ((n / 7) - 1) * 7 / 2;
//        System.out.println(sn);
        // 剩下的个数,也是一个等差数列
        int n2 = n % 7;
        int sn2 = (n2) * ((n / 7) + 1) + n2 * (n2 - 1) / 2;
        sum = sn + sn2;
        return sum;
    }
}
